메뉴

#GPU 커널 최적화

MP
MarkTechPost 55일 전
IMP 8

RightNow AI, 오픈소스 GPU 커널 최적화 프레임워크 'AutoKernel' 공개

RightNow AI 연구팀은 자율적 LLM 에이전트 루프를 활용해 임의의 PyTorch 모델에 대한 GPU 커널 최적화를 자동화하는 오픈소스 프레임워크인 'AutoKernel'을 공개했습니다. 이 프레임워크는 머신러닝 엔지니어링에서 가장 까다로운 작업 중 하나인 고성능 GPU 코드 작성을 완전히 자동화하는 것을 목표로 합니다. 사용자가 PyTorch 모델을 제공하면 AI가 자율적으로 코드를 최적화하는 접근 방식을 채택했습니다.

오픈소스 GPU 커널 최적화 PyTorch